package com.example.industrialpipelinemonitoring.utils;

import com.auth0.jwt.JWT;
import com.auth0.jwt.JWTCreator;
import com.auth0.jwt.algorithms.Algorithm;
import com.auth0.jwt.interfaces.DecodedJWT;

import java.util.Calendar;
import java.util.HashMap;
import java.util.Map;


/**
 * @author Dsy
 * @create 2020-11-29  21:56
 */
public class JwtUtils {
    // 设置sign（密钥）
    public static final String SIGN = "qweasd";

    /**
     * 通过uId生成token
     * @param uId
     * @return
     */
    public static String getTokenByUId(String uId){
        HashMap<String, String> map = new HashMap<>();
        map.put("uId", uId);
        return getToken(map);
    }

    /**
     * 根据token拿到uId
     * @param token
     * @return
     */
    public static String verifyTokenByUId(String token){
        DecodedJWT jwt = verify(token);
        String uId = jwt.getClaim("uId").asString();
        return uId;
    }

    /**
     * 生成token 形式为：header.payload.sign
     */
    public static String getToken(Map<String, String> map) {
        // 过期时间为七天
        Calendar instance = Calendar.getInstance();
        instance.add(Calendar.DATE, 7);
        // 创建JWT builder
        JWTCreator.Builder builder = JWT.create();
        // 存入信息payload中信息（也就是map中的信息）
        map.forEach((k, v) -> builder.withClaim(k, v));
        // 存入过期时间和密钥
        String token = builder.withExpiresAt(instance.getTime())
                .sign(Algorithm.HMAC256(SIGN));
        return token;
    }

    /**
     * 验证token 如果正确则返回token信息
     */
    public static DecodedJWT verify(String token) {
        // 如果token不对会直接抛出异常
        return JWT.require(Algorithm.HMAC256(SIGN)).build().verify(token);
    }

}